#!/usr/bin/env python
#
# Copyright 2007 Google Inc.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
#

import os
import cgi
import wsgiref.handlers
from google.appengine.ext import db
from google.appengine.api import users
from google.appengine.ext import webapp
from google.appengine.ext.webapp import template

class MainPage(webapp.RequestHandler):
  def get(self):
      template_values = {}
      path = os.path.join(os.path.dirname(__file__), 'index.html')
      self.response.out.write(template.render(path, template_values))

  def post(self):
      path = os.path.join(os.path.dirname(__file__), 'index.html')
      
      noter = db.GqlQuery("SELECT * FROM Note WHERE readkeyHash = :key", key = self.request.get('key'))
      notew = db.GqlQuery("SELECT * FROM Note WHERE writekeyHash = :key", key = self.request.get('key'))
      
      valuer = Note()
      valuew = Note()
      
      for note in noter:
          valuer = note

      for note in notew:
          valuew = note
          
      if valuer.readkeyHash == self.request.get('key'):
          mode = "read"
          template_values = {'note': repr(valuer.dataCrypted), 'mode': mode, 'key': self.request.get('key')}
          self.response.out.write(template.render(path, template_values))
      else:
          if valuew.writekeyHash == self.request.get('key'):
              mode = "write"
              template_values = {'note': repr(valuew.dataCrypted), 'mode': mode, 'key': self.request.get('key')}
              self.response.out.write(template.render(path, template_values))
          else:
              mode = "error"
              template_values = {'error': "Key not found", 'mode': mode}
              self.response.out.write(template.render(path, template_values))
              return
      
class AddNew(webapp.RequestHandler):
  def get(self):
      template_values = {}
      path = os.path.join(os.path.dirname(__file__), 'new.html')
      self.response.out.write(template.render(path, template_values))

      
class NewData(webapp.RequestHandler):
  def post(self):
      note = Note()
      note.readkeyHash = self.request.get('readkey')
      note.writekeyHash = self.request.get('writekey')
      note.dataCrypted = db.Blob(repr(self.request.get('information')))
      if(note.readkeyHash == note.writekeyHash):
          template_values = {}
          path = os.path.join(os.path.dirname(__file__), 'error.html')
          self.response.out.write(template.render(path, template_values))
          return
      note.put()
      template_values = {}
      path = os.path.join(os.path.dirname(__file__), 'done.html')
      self.response.out.write(template.render(path, template_values))

class Admin(webapp.RequestHandler):
    def get(self):
      notes = db.GqlQuery("SELECT * FROM Note")
      template_values = {'notes': notes}
      path = os.path.join(os.path.dirname(__file__), 'admin.html')
      self.response.out.write(template.render(path, template_values))

class Update(webapp.RequestHandler):
  def post(self):
      
      template_values = {}
      path = os.path.join(os.path.dirname(__file__), 'update.html')
      self.response.out.write(template.render(path, template_values))

class Note(db.Model):
  date = db.DateTimeProperty(auto_now_add=True)
  readkeyHash = db.StringProperty()
  writekeyHash = db.StringProperty()
  dataCrypted = db.BlobProperty()


def main():
  application = webapp.WSGIApplication(
                                       [('/', MainPage),
                                        ('/add', AddNew),
                                        ('/newdata', NewData),
                                        ('/update', Update),
                                        ('/admin', Admin)],
                                        debug=True)
  wsgiref.handlers.CGIHandler().run(application)

if __name__ == "__main__":
  main()
